package dao.test;

import java.util.ArrayList;
import java.util.Date;
import java.util.List;

import util.DateHelper;
import dao.TrajetDao;
import entity.Lieu;
import entity.ModeTransport;
import entity.Trajet;

public class TrajetTestDao implements TrajetDao {

	List<Trajet> listeTrajets = new ArrayList<Trajet>();

	public TrajetTestDao() {
		DateHelper date1 = new DateHelper(2008, 04, 1, 14, 30);
		DateHelper date2 = new DateHelper(2008, 04, 1, 15, 30);
		DateHelper date3 = new DateHelper(2008, 12, 15, 14, 30);
		DateHelper date4 = new DateHelper(2008, 12, 15, 15, 30);
		Lieu depart = new Lieu("PARIS");
		Lieu arrivee1 = new Lieu("BARCELONE");
		Lieu arrivee2 = new Lieu("MADRID");
		ModeTransport modeTransport = new ModeTransport("Avion");
		listeTrajets.add(new Trajet(date2.getDate(), date1.getDate(), depart,
				arrivee1, modeTransport, 300));
		listeTrajets.add(new Trajet(date4.getDate(), date3.getDate(), arrivee1,
				depart, modeTransport, 300));
		listeTrajets.add(new Trajet(date2.getDate(), date1.getDate(), depart,
				arrivee2, modeTransport, 350));
		listeTrajets.add(new Trajet(date4.getDate(), date3.getDate(), arrivee2,
				depart, modeTransport, 350));
	}

	@Override
	public List<Trajet> getAllTrajets() {
		return listeTrajets;
	}

	@Override
	public Trajet getTrajetById(long id) {
		return listeTrajets.get((int) id);
	}

	@Override
	public List<Trajet> getAllTrajetsDepuis(String lieuDepart) {
		List<Trajet> results = new ArrayList<Trajet>();
		for (Trajet trajet : listeTrajets) {
			if (trajet.getLieuDepart().getLieu().equals(lieuDepart))
				results.add(trajet);
		}
		return results;
	}

	@Override
	public List<Trajet> getTrajets(String lieuDepart, String lieuArrivee,
			Date allee, Date retour) {
		// TODO Auto-generated method stub
		return null;
	}

	
}
